home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / bin / apt-mark < prev    next >
Text File  |  2008-08-14  |  2KB  |  66 lines
  1. #!/usr/bin/python
  2.  
  3. from optparse import OptionParser
  4.  
  5. try:
  6.     import apt_pkg
  7. except ImportError:
  8.     print "Error importing apt_pkg, is python-apt installed?"
  9.     
  10. import sys
  11. import os.path
  12.     
  13. actions = { "markauto" : 1,
  14.             "unmarkauto": 0
  15.           }
  16.  
  17. if __name__ == "__main__":
  18.     apt_pkg.init()
  19.  
  20.     # option parsing
  21.     parser = OptionParser()
  22.     parser.usage = "%prog [options] {markauto|unmarkauto} packages..."
  23.     parser.add_option("-f", "--file", action="store", type="string",
  24.                       dest="filename",
  25.                       help="read/write a different file")
  26.     parser.add_option("-v", "--verbose",
  27.                       action="store_true", dest="verbose", default=False,
  28.                       help="print verbose status messages to stdout")
  29.     (options, args) = parser.parse_args()
  30.     if len(args) < 2:
  31.         parser.error("not enough argument")
  32.  
  33.     # get pkgs to change
  34.     if args[0] not in actions.keys():
  35.         parser.error("first argument must be 'markauto' or 'unmarkauto'")
  36.     pkgs = args[1:]
  37.     action = actions[args[0]]
  38.  
  39.     # get the state-file
  40.     if not options.filename:
  41.         STATE_FILE = apt_pkg.Config.FindDir("Dir::State") + "extended_states"
  42.     else:
  43.         STATE_FILE=options.filename
  44.  
  45.     # open the statefile
  46.     if os.path.exists(STATE_FILE):
  47.         tagfile = apt_pkg.ParseTagFile(open(STATE_FILE))
  48.         outfile = open(STATE_FILE+".tmp","w")
  49.         while tagfile.Step():
  50.             pkgname = tagfile.Section.get("Package")
  51.             autoInst = tagfile.Section.get("Auto-Installed")
  52.             if pkgname in pkgs:
  53.                 if options.verbose:
  54.                     print "changing %s to %s" % (pkgname,action)
  55.                 newsec = apt_pkg.RewriteSection(tagfile.Section,
  56.                                        [],
  57.                                        [ ("Auto-Installed",str(action)) ]
  58.                                        )
  59.                 outfile.write(newsec+"\n")
  60.             else:
  61.                 outfile.write(str(tagfile.Section)+"\n")
  62.         # all done, rename the tmpfile
  63.         os.chmod(outfile.name, 0644)
  64.         os.rename(outfile.name, STATE_FILE)
  65.         os.chmod(STATE_FILE, 0644)
  66.